package org.subside.List;

import org.subside.List.pojo.Person;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @Author : zhangbo
 * @data : 2024/5/16
 * @description :映射(map/flatMap)将一个流的元素按照一定的映射规则映射到另一个流中。分为map和flatMap
 */
public class StreamTest4 {
    /**
     * map：接收一个函数作为参数，该函数会被应用到每个元素上，并将其映射成一个新的元素。
     * flatMap：接收一个函数作为参数，将流中的每个值都换成另一个流，然后把所有流连接成一个流。
     */
    public static void main(String[] args) {
        //1.英文字符串数组的元素全部改为大写。整数数组每个元素+3。
        String[] strArr = { "abcd", "bcdd", "defde", "fTr" };
        List<String> strList = Arrays.stream(strArr).map(String::toUpperCase).collect(Collectors.toList());
        List<Integer> intList = Arrays.asList(1, 3, 5, 7, 9, 11);
        List<Integer> intListNew = intList.stream().map(x -> x + 3).collect(Collectors.toList());
        System.out.println("每个元素大写：" + strList);
        System.out.println("每个元素+3：" + intListNew);

        System.out.println("================================================");
        //2.将员工的薪资全部增加10000
        List<Person> personList = new ArrayList<Person>();
        personList.add(new Person("Tom", 8900, 23, "male", "NewYork"));
        personList.add(new Person("Jack", 7000, 25, "male", "Washington"));
        personList.add(new Person("Lily", 7800, 21, "female", "Washington"));
        personList.add(new Person("Anni", 8200, 24, "female", "NewYork"));
        personList.add(new Person("Owen", 9500, 25, "male", "NewYork"));
        personList.add(new Person("Alisa", 7900, 26, "female", "NewYork"));

        // 不改变原来员工集合的方式----不改变元对象的结构---new
        List<Person> personListNew = personList.stream().map(person -> {
            Person personNew = new Person(person.getName(), 0, 0, null, null);
            personNew.setSalary(person.getSalary() + 10000);
            return personNew;
        }).collect(Collectors.toList());

        for (Person person : personList) {
            System.out.println("一次改动前:"+person.getName() +"======"+ person.getSalary());
        }
        for (Person person : personListNew) {
            System.out.println("一次改动后:"+person.getName() +"======"+ person.getSalary());
        }
        // 改变原来员工集合的方式---直接改变元对象的数据
        List<Person> personListNew2 = personList.stream().map(person -> {
            person.setSalary(person.getSalary() + 10000);
            return person;
        }).collect(Collectors.toList());
        System.out.println("二次改动前：" + personList.get(0).getName() + "-->" + personList.get(0).getSalary());
        System.out.println("二次改动后：" + personListNew2.get(0).getName() + "-->" + personListNew2.get(0).getSalary());
    }
}
